# _*_ coding:utf-8 _*_
import os
import io
import json
import pymysql

def getAllGps():
    f = open("data/gps.txt")
    allGps = f.readlines()
    gpsList = []
    for gps in allGps:
        item = gps.replace("\n", "")
        info = item.split(" ")
        gpsInfo = {
            "city": info[2],
            "gps": [info[3], info[4]]
        }
        gpsList.append(gpsInfo)
    return gpsList

def getGpsByCity(city):
    allGps = getAllGps()
    for g in allGps:
        if city in g['city']:
            return g

def init():
    allCity = ['深圳', '成都', '重庆', '广州', '北京', '上海', '杭州', '西安', '昆明', '厦门', '南京', '哈尔滨', '上海', '青岛', '郑州', '沈阳', '济南', '贵阳', '海口', '长沙', '福州', '乌鲁木齐', '大连', '三亚', '南宁', '长春', '天津', '武汉', '南昌', '珠海', '合肥', '兰州', '太原', '银川', '温州', '包头', '宁波', '呼和浩特', '北京', '丽江', '西双版纳', '石家庄', '烟台', '西宁', '拉萨', '景德镇', '南通', '汕头', '桂林', '常州', '湛江', '阿克苏', '赤峰', '长治', '台州', '北海', '库尔勒', '乌兰浩特', '西昌', '喀什', '威海', '衢州', '柳州', '伊宁', '海拉尔', '连云港', '榆林', '徐州', '和田', '牡丹江', '芒市', '宜宾', '南阳', '大理', '梅县', '张家界', '宜昌', '黄山', '赣州', '通辽', '昌都', '延安', '武夷山', '佳木斯', '泸州', '舟山', '洛阳', '嘉峪关', '齐齐哈尔', '恩施', '敦煌', '南充', '保山', '延吉', '锡林浩特', '丹东', '锦州', '昭通', '安庆', '汉中', '且末', '秦皇岛', '塔城', '黑河', '潍坊', '富蕴', '朝阳', '石家庄']
    allGps = getAllGps()
    resList = []
    for c in allCity:
        g = getGpsByCity(c)
        resList.append(g)
    res = {}
    res = {
        "data": resList
    }
    print(len(resList))
    print(len(allCity))
    fo = open("data/city-gps.json", "w", encoding='utf-8')
    fo.write(json.dumps(res, ensure_ascii=False))
    # 关闭文件
    fo.close()

if __name__ == "__main__":
    init()